from openai import OpenAI
import sys
import os

# 添加项目根目录到Python路径
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))

from config_reader import get_siliconflow_config

# 获取Siliconflow配置
siliconflow_config = get_siliconflow_config()

client = OpenAI(api_key=siliconflow_config['api_key'],
                base_url=siliconflow_config['base_url'])

prompt = '''
##角色##
你是甄嬛的扮演者。

##任务##
跟用户进行角色扮演。你需要先判断出用户的身份，再模仿甄嬛的性格和语气做出回复。

##输出要求##
不超过50个字，不要带侧面描述。

##示例对话##
--1--
用户: "你是谁？",
甄嬛: "我是甄嬛，家父是大理寺少卿甄远道。"
--2--
用户："华妃娘娘命奴才给您送些礼物来。",
甄嬛："多谢娘娘美意，有劳公公向娘娘转达嫔妾的谢意。坐下喝杯茶吧。"
--3--
用户："莞姐姐忙了一日，我现在才敢过来，不算打扰吧？",
甄嬛："怎么会，本来正打算过去看妹妹呢，谁知妹妹到先来了，坐。浣碧，上茶。"

##用户的发言##
人如花，花亦如人，只有品性和婉的人，才会喜欢品性和婉的花。
'''

response_with_skill = client.chat.completions.create(
    model="Qwen/Qwen3-8B",
    messages=[
        {'role': 'user', 'content': prompt}
    ],
    max_tokens=1024,
    temperature=0.9,  # 温度系数，越接近 1 模型输出越随机，越接近 0 模型输出越固定
    stream=False
)

print(f"带有技巧的提问的结果：\n{response_with_skill.choices[0].message.content}")